package com.kamistoat.meimeistore.commonredis.configure;

import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONReader;
import com.alibaba.fastjson2.JSONWriter;
import org.springframework.data.redis.serializer.RedisSerializer;
import org.springframework.data.redis.serializer.SerializationException;

import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;

/**
 * 自定义的FastJSON序列化机制，将作为redis的序列化机制。对RedisTemplate和Redisson均生效
 *
 * @param <T>
 */
public class FastJson2JsonRedisSerializer<T> implements RedisSerializer<T> {

    // 默认字符编码为 UTF-8
    public static final Charset DEFAULT_CHARSET = StandardCharsets.UTF_8;

    // 不是final类型也可以。只是因为 clazz只会读取不会更改所以才写成 final
    private final Class<T> clazz;

    public FastJson2JsonRedisSerializer(Class<T> clazz) {
        super();
        this.clazz = clazz;
    }

    /**
     * 序列化函数
     *
     * @param t 任意类T的类实例
     * @return 序列化后的结果，注意最后要转成 byte[]
     * @throws SerializationException
     */
    @Override
    public byte[] serialize(T t) throws SerializationException {
        if (t == null) {
            return new byte[0];
        }
        // 同样使用 fastJson2.toJSONString()对类实例转型。增加额外参数，要求转型成JSON时保留元类全路径名称，方便反序列化
        // 然后再将 JSONString转成 byte[]
        return JSON.toJSONString(t, JSONWriter.Feature.WriteClassName).getBytes(DEFAULT_CHARSET);
    }

    /**
     * 反序列化函数
     *
     * @param bytes byte[]
     * @return 发序列化结果
     * @throws SerializationException
     */
    @Override
    public T deserialize(byte[] bytes) throws SerializationException {
        if (bytes == null || bytes.length <= 0) {
            return null;
        }
        // 先将 byte[]转成String
        String uni = new String(bytes, DEFAULT_CHARSET);
        // 同样使用 FastJSON2.parseObject()对String进行反转型。增加额外参数，要求支持自动转型
        return JSON.parseObject(uni, this.clazz, JSONReader.Feature.SupportAutoType);

    }
}
